package MonotonicStack;

import java.util.ArrayDeque;
import java.util.Arrays;
import java.util.Deque;
import java.util.Stack;

/**
 * 496.下一格更大元素|
 * nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x 大的元素。
 * 给你两个 没有重复元素 的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集。
 * 对于每个 0 <= i < nums1.length ，找出满足 nums1[i] == nums2[j] 的下标 j ，并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素，那么本次查询的答案是 -1 。
 * 返回一个长度为 nums1.length 的数组 ans 作为答案，满足 ans[i] 是如上所述的 下一个更大元素 。
 */
public class nextGreaterElement {
    /**
     * 一刷
     */
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Stack<Integer> stack = new Stack<>();
        int[] result = new int[nums1.length];
        Arrays.fill(result, -1);
        stack.push(0);
        for (int i = 1; i < nums2.length; i++) {
            if (nums2[i] <= nums2[stack.peek()]) {
                stack.push(i);
            } else {
                while (!stack.isEmpty() && nums2[stack.peek()] < nums2[i]) {
                    int index=find(nums1, nums2[stack.peek()]);
                    if(index==-1) break;
                    result[index] = nums2[i];
                    stack.pop();
                }
                stack.push(i);
            }
        }
        return result;
    }

    public int find(int[] nums, int target) {
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] == target) {
                return i;
            }
        }
        return -1;
    }
    public static void main(String[] args) {
        int[] nums1 = {1,3,5,2,4};
        int[] nums2 = {5,4,3,2,1};
        nextGreaterElement nextGreaterElement = new nextGreaterElement();
        System.out.println(Arrays.toString(nextGreaterElement.nextGreaterElement2(nums1, nums2)));
    }

    /**
     * 二刷
     */
    public int[] nextGreaterElement2(int[] nums1, int[] nums2) {
        int[] temp = new int[nums2.length];
        int[] res = new int[nums1.length];
        Deque<Integer> stack = new ArrayDeque<>();
        Arrays.fill(temp, -1);
        stack.push(0);
        for (int i = 1; i < nums2.length; i++) {
            while (!stack.isEmpty() && nums2[i] > nums2[stack.peek()]) {
                int index = stack.pop();
                temp[index] = nums2[i];
            }
            stack.push(i);
        }
        for (int i = 0; i < nums2.length; i++) {
            for (int j = 0; j < nums1.length; j++) {
                if(nums2[i]==nums1[j]){
                    res[j]=temp[i];
                }
            }
        }
        return res;
    }
}
